package com.yanxu;

import com.yanxu.domain.Event;
import org.apache.flink.api.common.functions.MapFunction;
import org.apache.flink.configuration.Configuration;
import org.apache.flink.streaming.api.datastream.DataStreamSource;
import org.apache.flink.streaming.api.datastream.SingleOutputStreamOperator;
import org.apache.flink.streaming.api.environment.StreamExecutionEnvironment;
import org.apache.flink.streaming.api.functions.sink.RichSinkFunction;

/**
 * @author 折戟沉沙铁未销
 * @version V1.0
 * @date 2025-07-13-2025
 * @Description: api 学习
 *  sink 自定义sink
 */
public class Api13_Sink_Custom {
    public static void main(String[] args) throws Exception {
        //获取执行环境
        StreamExecutionEnvironment env = StreamExecutionEnvironment.getExecutionEnvironment();
        env.setParallelism(1);

        //源 -算子
        // 使用 addSource 自定义数据源
        DataStreamSource<Event> streamSource = env.fromElements(new Event("Mary", "./home", 1000L),
                new Event("Bob", "./cart", 2000L),
                new Event("Alice", "./prod?id=100", 3000L),
                new Event("Alice", "./prod?id=200", 3500L),
                new Event("Bob", "./prod?id=2", 2500L),
                new Event("Alice", "./prod?id=300", 3600L),
                new Event("Bob", "./home", 3000L),
                new Event("Bob", "./prod?id=1", 2300L),
                new Event("Bob", "./prod?id=3", 3300L));

        // 使用map 算子进行处理
        SingleOutputStreamOperator<String> mapOperator = streamSource.map(new MapFunction<Event, String>() {
            @Override
            public String map(Event event) throws Exception {
                return event.getName();
            }
        });

        //自定义输出
        mapOperator.addSink(new CustomSink());

        //对环境进行执行。
        env.execute();
    }
}

class CustomSink extends RichSinkFunction<String> {
    // 最重要的方法 !!!!!!
    // 重写invoke方法，此方法在每个到来的数据元素上执行
    // 参数text表示接收到的数据，context用于访问Flink的运行时上下文
    @Override
    public void invoke(String value, Context context) throws Exception {
        System.out.println("my custom ......" + value);
    }

    // 重写open方法，当sink函数打开时调用，可进行一些初始化操作
    // 参数parameters提供sink函数配置信息
    @Override
    public void open(Configuration parameters) throws Exception {
        // 此处可以根据需要添加初始化代码，目前保持为空
    }


    // 重写close方法，当sink函数关闭时调用，可进行资源清理等操作
    @Override
    public void close() throws Exception {
        // 此处可以根据需要添加资源清理或关闭相关代码，目前保持为空
    }
}